#ifndef _THREAD_H_
#define _THREAD_H_

#include <iostream>
#include <string>
#include <pthread.h>
#include <cstdio>
#include <cstring>
#include <functional>

namespace ThreadModule
{
    // 计数器：给每个线程一个名字
    static uint32_t number = 1;

    template<typename T>
    class Thread
    {
        using func_t = std::function<void(T)>; // 暂时这样写
    private:
        // 开启分离
        void EnableDetach()
        {
            std::cout << "线程被分离了" << std::endl;
            _isdetach = true;
        }
        void EnableRunning()
        {
            _isrunning = true;
        }

        // 类内创建线程不能这样设置回调函数，因为成员函数第一个参数默认是this指针
        // 解决方法就是在前面加个static
        // 类内进行函数调用的方法
        static void *Routine(void *args)
        {
            Thread<T> *self = static_cast<Thread<T> *>(args);
            self->EnableRunning();
            if (self->_isdetach)
                self->Detach();

            self->_func(self->_data); // 类内访问函数
            return nullptr;
        }

    public:
        Thread(func_t func, T data) : _tid(0), _isdetach(false), _isrunning(false), res(nullptr), _func(func), _data(data)
        {
            _name = "thread-" + std::to_string(number++);
        }
        // 设置分离
        void Detach()
        {
            if (_isdetach)
                return;
            if (_isrunning)
                pthread_detach(_tid);
            EnableDetach();
        }

        bool Start()
        {
            if (_isrunning)
                return false;
            int n = pthread_create(&_tid, nullptr, Routine, this);
            if (n != 0)
            {
                std::cerr << "creat thread error: " << strerror(n) << std::endl;
                return false;
            }
            else
            {
                std::cout << _name << " creat success!" << std::endl;
                return true;
            }
        }
        bool Stop()
        {
            if (_isrunning)
            {
                int n = pthread_cancel(_tid);
                if (n != 0)
                {
                    std::cerr << "Stop thread error: " << strerror(n) << std::endl;
                    return false;
                }
                else
                {
                    _isrunning = false;
                    std::cout << _name << " stop" << std::endl;
                    return true;
                }
            }
            return false;
        }

        void Join()
        {
            if (_isdetach)
            {
                std::cout << "新线程已经设置分离，不能接收到了" << std::endl;
                return;
            }
            int n = pthread_join(_tid, &res);
            if (n != 0)
            {
                std::cerr << "Join thread error: " << strerror(n) << std::endl;
            }
            else
            {
                std::cout << "Join success!" << std::endl;
            }
        }
        ~Thread()
        {
        }

    private:
        // 线程属性
        pthread_t _tid;    // tid
        std::string _name; // 线程名字
        bool _isdetach;    // 是否分离
        bool _isrunning;   // 线程是否是运行起来了
        void *res;
        func_t _func; // 函数类型,我们需要指派的任务
        T _data;
    };
}

#endif